package com.carrot.mq.mqserver.core;

import com.carrot.mq.common.ConsumerEnv;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.core.type.TypeReference;
import com.fasterxml.jackson.databind.ObjectMapper;
import lombok.Data;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * @author 兴趣使然黄小黄
 * @version 1.0
 * @date 2023/7/31 18:52
 * 存储消息的队列
 */
@Data
public class MSGQueue {

    // 队列的身份标识
    private String name;

    // 队列是否持久化
    private boolean durable;

    // 是否独有(仅自己可用)
    private boolean exclusive;

    // 无消费者使用, 则自动删除
    private boolean autoDelete;

    // 额外参数选项, 存储数据库需要转 json 字符串
    private Map<String, Object> arguments = new HashMap<>();

    // 当前队列有哪些消费者订阅
    private List<ConsumerEnv> consumerEnvList = new ArrayList<>();

    // 记录当前取到了第几个消费者, 方便轮询策略的实现
    private AtomicInteger consumerSeq = new AtomicInteger(0);

    /**
     * 添加一个新的订阅者
     * @param consumerEnv 订阅者
     */
    public void addConsumerEnv(ConsumerEnv consumerEnv) {
        consumerEnvList.add(consumerEnv);
    }

    /**
     * 轮询的方式, 挑选一个订阅者, 用来处理当前的消息
     * @return 订阅者
     */
    public ConsumerEnv chooseConsumer() {
        if (consumerEnvList.size() == 0) {
            // 当前没有订阅者
            return null;
        }
        // 计算当前要取的元素下标
        int index = consumerSeq.get() % consumerEnvList.size();
        consumerSeq.getAndIncrement();
        return consumerEnvList.get(index);
    }

    public String getArguments() {
        // 是把当前的 arguments 参数, 从 Map 转成 String (JSON)
        ObjectMapper objectMapper = new ObjectMapper();
        try {
            return objectMapper.writeValueAsString(arguments);
        } catch (JsonProcessingException e) {
            e.printStackTrace();
        }
        // 如果代码真异常了, 返回一个空的 json 字符串就 ok
        return "{}";
    }

    // 从数据库中读数据构造对象时会自动调用
    public void setArguments(String argumentsJson) {
        // 把参数中的 argumentsJson 按照 JSON 格式解析, 转成
        // 上述的 Map 对象
        ObjectMapper objectMapper = new ObjectMapper();
        try {
            this.arguments = objectMapper.readValue(argumentsJson, new TypeReference<HashMap<String, Object>>() {});
        } catch (JsonProcessingException e) {
            e.printStackTrace();
        }
    }

    // 在这里针对 arguments, 再提供一组 getter setter , 用来去更方便的获取/设置这里的键值对.
    // 这一组在 java 代码内部使用 (比如测试的时候)
    public Object getArguments(String key) {
        return arguments.get(key);
    }

    public void setArguments(String key, Object value) {
        arguments.put(key, value);
    }

    public void setArguments(Map<String, Object> arguments) {
        this.arguments = arguments;
    }
}
